# !/usr/bin/python3
# -*- coding: utf-8 -*-

"""
租户信息表模型类
----------------------------------------------------
@Project :   xinhou-openai-framework
@File    :   Tenant.py
@Contact :   sp_hrz@qq.com

@Modify Time      @Author    @Version    @Desciption
------------      -------    --------    -----------
2023/04/16 22:04  peng.shen   v1.0.0     None
"""

from datetime import datetime
from typing import Optional

from pydantic import BaseModel, Field


class ReqTenantFindSchema(BaseModel):
    """
    租户信息表 产线入参参数&校验模型
    """

    id: Optional[int] = Field(
        default=None, title="主键ID",
    )

    tenant_code: Optional[str] = Field(
        default=None, title="租户编码",
    )

    tenant_name: Optional[str] = Field(
        default=None, title="租户名称",
    )

    api_key: Optional[str] = Field(
        default=None, title="租户ApiKey",
    )

    num_tokens: Optional[int] = Field(
        default=None, title="api调用计算token总数",
    )
    status: Optional[int] = Field(
        default=None, title="状态:1=正常,2=停用"
    )
    create_by: Optional[str] = Field(
        default=None, title="创建者",
    )

    created_at: Optional[datetime] = Field(
        default=None, title="创建时间",
    )

    update_by: Optional[str] = Field(
        default=None, title="更新者",
    )

    updated_at: Optional[datetime] = Field(
        default=None, title="更新时间",
    )

    remark: Optional[str] = Field(
        default=None, title="备注",
    )


class ReqTenantSaveSchema(BaseModel):
    """
    租户信息表 保存入参参数&校验模型
    """

    tenant_code: Optional[str] = Field(
        default=None, title="租户编码",
    )

    tenant_name: Optional[str] = Field(
        default=None, title="租户名称",
    )

    api_key: Optional[str] = Field(
        default=None, title="租户ApiKey",
    )

    num_tokens: Optional[int] = Field(
        default='0', title="api调用计算token总数",
    )
    status: Optional[int] = Field(
        default='1', title="状态:1=正常,2=停用",
    )
    create_by: Optional[str] = Field(
        default=None, title="创建者",
    )

    created_at: Optional[datetime] = Field(
        default=None, title="创建时间",
    )

    update_by: Optional[str] = Field(
        default=None, title="更新者",
    )

    updated_at: Optional[datetime] = Field(
        default=None, title="更新时间",
    )

    remark: Optional[str] = Field(
        default=None, title="备注",
    )


class ReqTenantUpdateSchema(BaseModel):
    """
    租户信息表 更新入参参数&校验模型
    """

    id: int = Field(
        title="主键ID",
    )

    tenant_code: Optional[str] = Field(
        default=None, title="租户编码",
    )

    tenant_name: Optional[str] = Field(
        default=None, title="租户名称",
    )

    api_key: Optional[str] = Field(
        default=None, title="租户ApiKey",
    )

    num_tokens: Optional[int] = Field(
        default=None, title="api调用计算token总数",
    )
    status: Optional[int] = Field(
        default=None, title="状态:1=正常,2=停用",
    )
    create_by: Optional[str] = Field(
        default=None, title="创建者",
    )

    created_at: Optional[datetime] = Field(
        default=None, title="创建时间",
    )

    update_by: Optional[str] = Field(
        default=None, title="更新者",
    )

    updated_at: Optional[datetime] = Field(
        default=None, title="更新时间",
    )

    remark: Optional[str] = Field(
        default=None, title="备注",
    )


class ResTenantDetailSchema(BaseModel):
    """
    租户信息表 出参参数&校验模型
    """

    id: Optional[int] = Field(
        default=None, title="主键ID",
    )

    tenant_code: Optional[str] = Field(
        default=None, title="租户编码",
    )

    tenant_name: Optional[str] = Field(
        default=None, title="租户名称",
    )

    api_key: Optional[str] = Field(
        default=None, title="租户ApiKey",
    )

    num_tokens: Optional[int] = Field(
        default=None, title="api调用计算token总数",
    )
    status: Optional[int] = Field(
        default=None, title="状态:1=正常,2=停用"
    )
    create_by: Optional[str] = Field(
        default=None, title="创建者",
    )

    created_at: Optional[datetime] = Field(
        default=None, title="创建时间",
    )

    update_by: Optional[str] = Field(
        default=None, title="更新者",
    )

    updated_at: Optional[datetime] = Field(
        default=None, title="更新时间",
    )

    remark: Optional[str] = Field(
        default=None, title="备注",
    )
